package com.zr.controller;

import java.io.IOException;
import java.io.PrintWriter;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

import com.zr.service.TreeDataService;
import com.zr.service.impl.TreeDataServiceImpl;
import com.zr.util.ChatWebSocket;
import com.zr.util.Hangye;

import net.sf.json.JSONArray;
/**
 * 取树形数据，生成树形目录的controller层
 * @author Administrator
 *
 */
public class TreeDataController extends HttpServlet{

	TreeDataService tds = new TreeDataServiceImpl();
	
	@Override
	protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
		doPost(req, resp);
	}
	
	@Override
	protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
		
		req.setCharacterEncoding("utf-8");
		resp.setCharacterEncoding("utf-8");
		
		JSONArray jArray = tds.getTreeDataById(-1);
		HttpSession session = req.getSession();
		
		if (((int)session.getAttribute("flag")) == 0) {
			Hangye hangye = new Hangye();
			hangye.start();
		}
		
		session.setAttribute("flag", (((int)session.getAttribute("flag"))+1));
		
		PrintWriter pWriter = resp.getWriter();
		pWriter.write(jArray.toString());
		
		pWriter.flush();
		pWriter.close();
		
	}
	
}
